import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 6], [7, 8]])

numpy_a = a.numpy() # 调用张量的numpy()方法，把张量转换为numpy数组
print(numpy_a)

# 1.逐元素加法
add = a +b

# 1.逐元素乘法
mul = a * b

# 2.矩阵乘法
matmul = a @ b

# 其他运算
sqrt = tf.sqrt(tf.cast(a, tf.float32))  # 平方根(需要转换为浮点型)
